export CUDA_VISIBLE_DEVICES=0
if [[ $3 == *"lbf"* ]]; then
    python3 src/main.py --eval --config=vdn --env-config=lbf4 --seed=1 --name=$3 --save_replay=True --save_traj=True with runner=episode_xp evaluate=True test_nepisode=50 checkpoint_path_ego=$1 checkpoint_path_tm=$2
elif [[ $3 == *"pp"* ]]; then
    python3 src/main.py --eval --config=vdn --env-config=pp --seed=1 --name=$3 --save_replay=True --save_traj=True with runner=episode_xp evaluate=True test_nepisode=50 checkpoint_path_ego=$1 checkpoint_path_tm=$2
elif [[ $3 == *"sc2"* ]]; then
    python3 src/main.py --eval --config=vdn --env-config=sc2_v2 --seed=1 --name=$3 --save_replay=False --save_traj=False with runner=episode_xp evaluate=True test_nepisode=50 checkpoint_path_ego=$1 checkpoint_path_tm=$2
else
    echo "No matching pattern found in the name."
fi